Depth first search algorithm in python